# 0~4之间的数值进行转换，当大于3时求平方，否则输出原大小。

count = 0
while count < 5:
    if count > 3:
        print(count ** 2)
    else:
        print(count)
    count = count + 1